clear all;
clc;

data = 'train';

load(['mnist_',data]);

for i = 0:9
    idx = find(yL == i);
    eval([data,int2str(i),'=xL(:,idx);']);
end

save(['mnist_split_',data],[data,'0'],[data,'1'],[data,'2'],[data,'3'],[data,'4'],[data,'5'],[data,'6'],[data,'7'],[data,'8'],[data,'9']);